#
#
import torch
import torch.nn as nn
import torch.utils.data as datautils

import torchvision
import torchvision.datasets as datasets
import torchvision.transforms as transforms

# from torchvision.transforms import (ToTensor, Normalize)

ROOT = "E:/sfxData/DeepLearning"

transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])

batch_size = 4

# train
trainset = datasets.CIFAR10(
    root=ROOT,
    train=True,
    download=True,
    transform=transform)
